﻿# -*- coding: utf-8 -*-
"""
1、实现一个函数获取一个目录下所有以.py 结尾的文件（包含目录下的子目录中的.py 文件，不要使用 glob 方法进行匹配）
2、实现一个可迭代的类
"""
# 第一题------->实现一个函数获取一个目录下所有以.py 结尾的文件（包含目录下的子目录中的.py 文件，不要使用 glob 方法进行匹配）
from pathlib import Path


def list_pyfile(path):
    path = Path(path)
    # for i in filter(lambda x:x.name.endswith('.py'),path.iterdir()):
    for i in path.iterdir():
        if i.is_dir():
            list_pyfile(str(i))
        elif i.name.endswith('.py'):
            print(i)


# 第二题----->实现一个可迭代的类
class Iter:
    def __init__(self, x):
        self.x = x

    def __iter__(self):
        return self

    def __next__(self):
        self.x += 1
        return self.x


if __name__ == '__main__':
    list_pyfile('c:/Temp')
    print('=' * 30)
    a = Iter(2)
    print(next(a), '----')
    for i, v in enumerate(a):
        print(v)
        if i == 10:
            break

# 完成的不错